package com.example.viewdemo.View

import android.app.Activity
import android.content.Context
import android.util.AttributeSet
import android.view.LayoutInflater
import android.widget.Button
import android.widget.LinearLayout
import android.widget.Toast
import com.example.viewdemo.R


class TitleLayout(context: Context?, attrs: AttributeSet?) : LinearLayout(context, attrs) {
    private lateinit var titleBack: Button
    private lateinit var titleEdit: Button

    init {
        LayoutInflater.from(context).inflate(R.layout.title, this)

        // 通过ID找到按钮，并设置点击事件
        titleBack = findViewById(R.id.titleBack) // 确保你的title.xml中有一个id为titleBack的按钮
        titleEdit = findViewById(R.id.titleEdit) // 确保你的title.xml中有一个id为titleEdit的按钮

        titleBack.setOnClickListener {
            // Kotlin 中的类型强制转换使用的关键字是 as
            val activity = context as? Activity
            activity?.finish() // 使用安全调用以避免对null调用finish()
        }
        titleEdit.setOnClickListener {
            Toast.makeText(context, "You clicked Edit button.", Toast.LENGTH_SHORT).show()
        }
    }
}